package first_practice;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * @author ZhiWen Ren
 * @version 1.0
 * @description: LeetCode-145-二叉树的后序遍历
 * @date 2025/4/12 16:13
 */
public class Solution145 {
    //    public List<Integer> postorderTraversal(TreeNode root) {
//        // 左 右 根
//        // 递归实现
//        List<Integer> ans = new ArrayList<>();
//        postOrder(root, ans);
//        return ans;
//    }
//
//    private void postOrder(TreeNode root, List<Integer> ans) {
//        if (root == null)
//            return;
//        postOrder(root.left, ans);
//        postOrder(root.right, ans);
//        ans.add(root.val);
//    }
    public List<Integer> postorderTraversal(TreeNode root) {
        // 迭代栈实现
        List<Integer> ans = new ArrayList<>();
        Deque<TreeNode> stack = new LinkedList<>();
        TreeNode prev = null;

        while (!stack.isEmpty() || root != null) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            if (root.right == null || root.right == prev) { //后序遍历需要考虑右节点
                ans.add(root.val);
                prev = root;
                root = null;
            } else {
                // 有右节点，root你还是重新回到栈里吧
                stack.push(root);
                root = root.right; // 继续将右节点及其下面的左节点压入栈中
            }
        }

        return ans;
    }
}
